package com.hc.programming.string;

import java.util.Arrays;

/**
 * 给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。如果 needle 不是 haystack 的一部分，则返回  -1 。
 * <p>
 * 示例 1：
 * 输入：haystack = "sadbutsad", needle = "sad"
 * 输出：0
 * 解释："sad" 在下标 0 和 6 处匹配。
 * 第一个匹配项的下标是 0 ，所以返回 0 。
 * 示例 2：
 * 输入：haystack = "leetcode", needle = "leeto"
 * 输出：-1
 * 解释："leeto" 没有在 "leetcode" 中出现，所以返回 -1 。
 * <p>
 * 提示：
 * 1 <= haystack.length, needle.length <= 10^4
 * haystack 和 needle 仅由小写英文字符组成
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/10/28 15:59
 */
public class 找出字符串中第一个匹配项的下标 {
    public static void main(String[] args) {
        System.out.println(strStr("sadbutsad", "sad") + " --- 0");
        System.out.println(strStr("leetcode", "leeto") + " --- -1");
        System.out.println(strStr("mississippi", "issip") + " --- 4");
        System.out.println(strStr("abababc", "ababc") + " --- 2");
        System.out.println(strStr("aabaaabaaac", "aabaaac") + " --- 4");
    }

    public static int strStr(String haystack, String needle) {
        return kmp(haystack, needle);
//        return haystack.indexOf(needle);
    }

    private static int kmp(String haystack, String needle) {
        int[] next = getNext(needle);
        System.out.println(Arrays.toString(next));
        int i = 0, j = 0;
        while ( i < haystack.length() && j < needle.length()) {
            if (haystack.charAt(i) == needle.charAt(j)) {
                j++;
                i++;
            } else if (j > 0) {
                j = next[j - 1];
            } else {
                i++;
            }
        }
        if (j == needle.length()) {
            return i - j;
        }
        return -1;
    }

    private static int[] getNext(String needle) {
        int[] next = new int[needle.length()];
        int j = 0; // 前缀末尾
        next[0] = 0;
        // i 后缀末尾
        for (int i = 1; i < needle.length(); i++) {
            while (j > 0 && needle.charAt(i) != needle.charAt(j)) {
                j = next[j - 1];
            }
            if (needle.charAt(i) == needle.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
